package com.catazzz.springbootstudy.common;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;

/**
 * @作者: Tao
 * @日期: 2024/12/30
 * @时间: 13:54
 * @描述: 填充策略 非严格模式<br><pre>
 *     严格模式：仅当属性值为null时，才会进行填充。如果字段已经有值，则不会覆盖原有值
 *          strictInsertFill()
 *     非严格模式：无论字段是否有值，都会进行自动填充，可能会覆盖已有值
 *          setFieldValByName()
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

//    /**
//     * 新增时填充策略（严格模式）
//     *
//     * @param metaObject 填充对象
//     */
//    @Override
//    public void insertFill(MetaObject metaObject) {
//        this.strictInsertFill(metaObject, "creator", String.class, getCurrentUser());
//        this.strictInsertFill(metaObject, "modifier", String.class, getCurrentUser());
//        this.strictInsertFill(metaObject, "createTime", LocalDateTime.class, LocalDateTime.now());
//        this.strictInsertFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
//    }
//
//    /**
//     * 更新时填充策略（严格模式）
//     *
//     * @param metaObject 填充对象
//     */
//    @Override
//    public void updateFill(MetaObject metaObject) {
//        this.strictUpdateFill(metaObject, "modifier", String.class, getCurrentUser());
//        this.strictUpdateFill(metaObject, "updateTime", LocalDateTime.class, LocalDateTime.now());
//    }

    /**
     * 新增时填充策略（非严格模式）
     *
     * @param metaObject 填充对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        this.setFieldValByName("creator", getCurrentUser(), metaObject);
        this.setFieldValByName("modifier", getCurrentUser(), metaObject);
        this.setFieldValByName("createTime", LocalDateTime.now(), metaObject);
        this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
    }

    /**
     * 更新时填充策略（非严格模式）
     *
     * @param metaObject 填充对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        this.setFieldValByName("modifier", getCurrentUser(), metaObject);
        this.setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
    }

    private String getCurrentUser() {
        return "管理员" + (int) (Math.random() * 10);
    }
}
